from app import app, db
from models.models import DepositBusiness

def init_deposit_business():
    with app.app_context():
        # 清空现有数据
        DepositBusiness.query.delete()
        
        # 添加存款业务数据
        businesses = [
            {
                'business_name': '活期',
                'description': '无固定存期，可随时存取，存取金额不限的一种比较灵活的存款',
                'min_amount': 0.0,
                'max_amount': None,
                'interest_rate': 0.35,
                'term_options': '',
                'is_active': 1
            },
            {
                'business_name': '定活两便',
                'description': '事先不约定存期，一次性存入，一次性支取的存款',
                'min_amount': 50.0,
                'max_amount': None,
                'interest_rate': 1.0,
                'term_options': '',
                'is_active': 1
            },
            {
                'business_name': '整存整取',
                'description': '选择存款期限，整笔存入，到期提取本息，是一种定期储蓄。银行提供的存款期限有1年、2年和3年',
                'min_amount': 1000.0,
                'max_amount': None,
                'interest_rate': 2.75,
                'term_options': '1年,2年,3年',
                'is_active': 1
            },
            {
                'business_name': '零存整取',
                'description': '一种事先原定金额，逐月按约定金额存入，到期支取本息的定期储蓄。银行提供的存款期限有1年、2年和3年',
                'min_amount': 50.0,
                'max_amount': None,
                'interest_rate': 1.35,
                'term_options': '1年,2年,3年',
                'is_active': 1
            }
        ]
        
        for business_data in businesses:
            business = DepositBusiness(**business_data)
            db.session.add(business)
        
        try:
            db.session.commit()
            print("存款业务数据初始化成功！")
            print("已添加以下业务类型：")
            for business in businesses:
                print(f"- {business['business_name']}: {business['description']}")
        except Exception as e:
            db.session.rollback()
            print(f"初始化失败：{e}")

if __name__ == '__main__':
    init_deposit_business() 